package com.yincb.helper;

import com.yincb.constans.DataSourceConstants;

public class DynamicDataSourceHelper {


    /**
     * 动态数据源名称上下文
     */
    private static final ThreadLocal<String> DATASOURCE_KEY = new ThreadLocal<>();

    /**
     * 设置/切换数据源
     */
    public static void setDynaDBKey(String key) {
        DATASOURCE_KEY.set(key);
    }

    /**
     * 获取数据源名称
     */
    public static String getDynaDBKey() {
        String key = DATASOURCE_KEY.get();
        return key == null ? DataSourceConstants.DS_KEY_YCB01 : key;
    }

    /**
     * 删除当前数据源名称
     */
    public static void removeDynaKey() {
        DATASOURCE_KEY.remove();
    }


}
